package logger

import (
	"os"

	"github.com/sirupsen/logrus"
	"github.com/spf13/viper"
)

var logger = logrus.New()

func init() {
	logger.Out = os.Stdout

	// 读取配置文件
	viper.SetConfigName("config")
	viper.SetConfigType("yaml")
	viper.AddConfigPath(".")
	viper.ReadInConfig()

	lv := viper.GetString("log.level")

	switch lv {
	case "dev":
		// 开发
		logger.SetLevel(logrus.DebugLevel)
	case "test":
		// 测试
		logger.SetLevel(logrus.InfoLevel)
	case "prod":
		// 生产
		logger.SetLevel(logrus.ErrorLevel)
	default:
		// 默认
		logger.SetLevel(logrus.InfoLevel)
	}

}

func Info(args ...interface{}) {
	logger.Info(args...)
}

func Debug(args ...interface{}) {
	logger.Debug(args...)
}

func Warn(args ...interface{}) {
	logger.Warn(args...)
}

func Error(args ...interface{}) {
	logger.Error(args...)
}
